#本脚本的目的是为了让一个文件夹下的图片统一裁剪到某一尺寸上

import argparse
from pathlib import Path
import cv2
from tqdm import tqdm

def Parse_Arguments():
    parser = argparse.ArgumentParser(description="")
    parser.add_argument('--img-dir', default=r"D:\tempDownload\dataset\elevator\temp", type=str)
    return parser.parse_args()

def main():
    args = Parse_Arguments()
    img_dir = Path(args.img_dir).glob("*.jpg")
    for img_path in tqdm(img_dir, desc="正在飞速的裁剪图片(●'◡'●)"):
        img = cv2.imread(str(img_path))
        crop_img = img[0:1320,:,:]
        cv2.imwrite(str(img_path), crop_img)
        # 测试
        # cv2.imwrite("a.jpg", crop_img)
        # break
if __name__ == "__main__":
    main()
